import cn.hutool.db.Db;
import cn.hutool.db.Entity;
import org.junit.Test;

import java.sql.SQLException;

/**
 * @Author: HAPPY
 * @Project_name: project
 * @Package_name: PACKAGE_NAME
 * @Date: 2021/11/18 23:09
 * @Description：
 */
public class AddPoster {
    // 为电影表添加poster字段，电影id + .jpg
    @Test
    public void test01() {
        try {
            for (Entity id : Db.use().query("select id from movie")) {
                Db.use().update(
                        Entity.create().set("poster", id.getInt("id") + ".jpg"), //修改的数据
                        Entity.create("movie").set("id", id.getInt("id"))  //where条件
                );
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }

    }
    // 为员工表添加poster字段，员工id + .jpg
    @Test
    public void test02() {
        try {
            for (Entity id : Db.use().query("select id from staff")) {
                Db.use().update(
                        Entity.create().set("avatar", id.getInt("id") + ".jpg"), //修改的数据
                        Entity.create("staff").set("id", id.getInt("id"))  //where条件
                );
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }

    }
}
